<template>
  <div class="editor">
    <div class="editor-header">
      <Header @editGraphConfig="onEditGraphConfig"
              @toExport="toExport"
              @toImport="toImport"
              @toClear="toClear">
      </Header>
    </div>
    <div class="editor-content">
      <div class="editor-sider">
        <Sider ref="siderRef" @startDrag="onStartDrag"></Sider>
      </div>
      <div class="editor-canvas">
        <FlowCanvas ref="flowCanvasRef" :dndContainer="siderRef"></FlowCanvas>
        <a-modal v-model:open="state.isExportDialogOpen" title="导出" @cancel="state.isExportDialogOpen = false">
          <a-textarea :rows="10" v-model:value="state.exportData" />
          <template #footer>
            <a-button type="primary" @click="handleCopyExport">复制</a-button>
          </template>
        </a-modal>
        <a-modal v-model:open="state.isImportDialogOpen" title="导入" @ok="handleImport"
          @cancel="state.isImportDialogOpen = false">
          <a-divider orientation="left">1.黏贴内容</a-divider>
          <a-textarea :rows="10" v-model:value="state.importData" />
          <a-divider orientation="left">2.选择布局</a-divider>
          <a-select v-model:value="state.layoutType">
            <a-select-option value="TB">从上到下</a-select-option>
            <a-select-option value="LR">从左到右</a-select-option>
          </a-select>
        </a-modal>
      </div>
    </div>
  </div>
</template>

<script setup>
import { ref, reactive } from 'vue';
import Header from './editor/Header.vue';
import Sider from './editor/Sider.vue';
import FlowCanvas from './editor/Canvas.vue';
import * as utils from '@/utils/index.js'
import yamlUtils from 'js-yaml'
import { notification } from 'ant-design-vue';

const flowCanvasRef = ref(null); // 画布容器的引用
const siderRef = ref(null); // 侧边栏容器的引用

const state = reactive({
  isExportDialogOpen: false, // 导出对话框的状态
  exportData: '', // 导出的数据
  importType: 'json',
  layoutType: 'TB', // 布局类型
})

const onStartDrag = ({ e, nodeType }) => {
  flowCanvasRef.value.onSiderStartDrag(e, nodeType); // 调用画布容器的方法
}
const onEditGraphConfig = () => {
  flowCanvasRef.value.onEditGraphConfig();
}

const toExport = (type) => {
  // 组只数据，将antv-x6格式转换为solon-flow格式
  const data = flowCanvasRef.value.getData();
  const nodeLinkMap = {}; // 用于存储节点和边的关联关系
  const nodes = [];
  let nodeEnd = null;

  data.graphViewData.cells.forEach(cell => {
    if (cell.shape == 'flow-edge') {
      const edgeData = cell.data || {}
      const edge = {
        v_source: cell.source.cell,
        v_sourcePort: cell.source.port,
        v_target: cell.target.cell,
        v_targetPort: cell.target.port,

        nextId: cell.target.cell, // 边的目标节点
        id: cell.id,
        title: edgeData.title,
        condition: edgeData.condition
      }
      nodeLinkMap[cell.source.cell] = nodeLinkMap[cell.source.cell] || []; // 初始化节点的边数组
      nodeLinkMap[cell.source.cell].push(edge); // 将边添加到节点的边数组中
    } else {
      const nodeData = cell.data || {};
      const node = {
        id: cell.id,
        type: nodeData.type,
        title: nodeData.title,
        v_x: cell.position.x, // 节点的 x 坐标
        v_y: cell.position.y, // 节点的 y 坐标
      };

      //简化导入（空内容可不出）
      if (nodeData.task) {
        node.task = nodeData.task;
      }

      if (nodeData.when) {
        node.when = nodeData.when;
      }

      if (nodeData.meta && Object.keys(nodeData.meta).length > 0) {
        node.meta = nodeData.meta;
      }

      //排序（确保 start 在最前）
      if (node.type == 'start') {
        nodes.unshift(node); //插到前面
      } else if (node.type == 'end') {
        nodeEnd = node;
      } else {
        nodes.push(node);
      }
    }
  })

  //排序（确保 end 在最后）
  if (nodeEnd) {
    nodes.push(nodeEnd);
  }

  nodes.forEach(node => {
    let link = nodeLinkMap[node.id];
    if (link) {
      //简化输出
      if (link instanceof Array) {
        if (link.length > 0) {
          node.link = link; // 将边数组添加到节点对象中
        }
      } else {
        node.link = link; // 将边数组添加到节点对象中
      }
    }
  });

  const graphData = data.graphData; // 构建最终的 JSON 数据
  graphData.layout = nodes

  console.log('graphData', graphData)
  if ('json' == type) {
    state.exportData = JSON.stringify(graphData, null, 4); // 格式化输出 JSON 数据
  } else {
    state.exportData = yamlUtils.dump(graphData); // 格式化输出 YAML 数据
  }

  state.isExportDialogOpen = true; // 打开导出对话框
}

function toImport(type) {
  state.importType = type
  state.isImportDialogOpen = true; // 打开导入对话框
}

function handleImport() {
  let dirType = state.layoutType // TB 从上到下 LR 从左到右
  let data = null
  let isAutoLayout = false;// 是否自动布局
  if (state.importType == 'json') {
    data = JSON.parse(state.importData); // 解析导入的数据
  } else {
    data = yamlUtils.load(state.importData); // 解析导入的数据
  }

  let portPosDef = ['port_b1', 'port_t1']
  if (dirType == 'LR') {
    portPosDef = ['port_r1', 'port_l1']
  }

  flowCanvasRef.value.clear(false); // 清空画布容器中的内容
  flowCanvasRef.value.setGraph(data); //
  // 组织数据，将solon-flow格式转换为antv-x6格式

  let layoutNodes = data.layout || data.nodes; //早期有配置是 nodes
  if (layoutNodes) {
    const graphViewData = {
      cells: [], // 存储节点和边的数组
    };

    let preNode = null; // 记录上一个节点

    let temp_x = 10;
    let temp_y = 10;
    if (typeof layoutNodes[0].v_x == 'undefined' || typeof layoutNodes[0].v_y == 'undefined' || layoutNodes[0].v_x == null || layoutNodes[0].v_y == null) {
      isAutoLayout = true;
    }

    //排序（确保 start 在最前）
    const nodes = [];
    let nodeEnd = [];
    layoutNodes.forEach(node => {
      if (node.type == 'start') {
        nodes.unshift(node); //插到前面
      } else if (node.type == 'end') {
        nodeEnd.push(node);
      } else {
        if (node.type == 'loop') {
          node.type = 'iterator';
        }
        nodes.push(node);
      }
    });

    //排序（确保 end 在最后）
    if (nodeEnd.length > 0) {
      nodes.push(...nodeEnd);
    }

    nodes.forEach(node => {
      node.type = node.type || 'activity'
      node.id = node.id || 'node_' + utils.uuid2()
      const nodeData = {
        id: node.id, // 节点的唯一标识符
        shape: node.type, // 节点的形状
        data: { // 节点的自定义数据
          id: node.id, // 节点的唯一标识符
          type: node.type, // 节点的类型
          title: node.title, // 节点的标题
          task: node.task, // 节点的任务
          when: node.when, // 节点的条件
          meta: node.meta, // 节点的元数据
        },
        position: { // 节点的位置
          x: node.v_x, // 节点的 x 坐标
          y: node.v_y, // 节点的 y 坐标
        },
      }

      if (typeof node.v_x == 'undefined' || typeof node.v_y == 'undefined' || node.v_x == null || node.v_y == null) {
        nodeData.position = {
          x: temp_x, // 节点的 x 坐标
          y: temp_y, // 节点的 y 坐标
        }
        temp_x += 100;
        temp_y += 100;
      }

      graphViewData.cells.push(nodeData); // 将节点数据添加到数组中

      if (node.link) {
        if (Array.isArray(node.link)) {
          node.link.forEach(link => {
            if (typeof link == 'object') {
              if (!link.id) {
                link.id = 'edge_' + utils.uuid2()
              }
              if (!link.v_source) {
                link.v_source = node.id,
                  link.v_sourcePort = portPosDef[0]
              }
              if (!link.v_target) {
                link.v_target = link.nextId,
                  link.v_targetPort = portPosDef[1]
              }
              const edgeData = {
                id: link.id, // 边的唯一标识符
                shape: 'flow-edge', // 边的形状
                source: { cell: link.v_source, port: link.v_sourcePort }, // 边的源节点和端口
                target: { cell: link.v_target, port: link.v_targetPort }, // 边的目标节点和端口
                data: { // 边的自定义数据
                  id: link.id, // 边的唯一标识符
                  nextId: link.target, // 边的目标节点
                  title: link.title, // 边的标题
                  condition: link.when || link.condition, //支持 when 配置（未来替代 condition）
                },
              }
              if (link.title) {
                edgeData.labels = [link.title]
              } else if (link.condition) {
                edgeData.labels = [link.condition]
              }
              graphViewData.cells.push(edgeData); // 将边数据添加到数组中
            } else {// string的情况
              const edgeData = buildEdgeForStringType(node.id, link, portPosDef) // 构建边数据的函数
              graphViewData.cells.push(edgeData); // 将边数据添加到数组中
            }


          })
        } else {
          // string的情况
          const edgeData = buildEdgeForStringType(node.id, node.link, portPosDef) // 构建边数据的函数
          graphViewData.cells.push(edgeData); // 将边数据添加到数组中
        }
      }

      if (preNode && (!preNode.link || preNode.link.length == 0)) {
        const cell = preNode
        const nextCell = node

        // 如果cell和nextCell都是end，不连接
        if (cell.type == 'end' && nextCell.type == 'end') {
            preNode = node;
            // 两个节点类型都为 end 时，跳过当前循环
            return;
        }
        const edgeData = buildEdgeForStringType(cell.id, nextCell.id, portPosDef) // 构建边数据的函数
        graphViewData.cells.push(edgeData);
      }

      preNode = node;
    })

    console.log('graphViewData', graphViewData)
    flowCanvasRef.value.setData(graphViewData); // 将节点和边数据设置到画布容器中
    if (isAutoLayout) {
      flowCanvasRef.value.autoLayout(dirType); // 自动缩放画布容器以适应所有节点和边
    }
  }


  state.importData = '';
  state.isImportDialogOpen = false; // 关闭导入对话框
}

function toClear() {
  flowCanvasRef.value.clear(); // 清空画布容器中的内容
}

function buildEdgeForStringType(source, target, portPosDef) {
  const edgeId = 'edge_' + utils.uuid2()

  const edgeData = {
    id: edgeId, // 边的唯一标识符
    shape: 'flow-edge', // 边的形状
    source: { cell: source, port: portPosDef[0] }, // 边的源节点和端口
    target: { cell: target, port: portPosDef[1] }, // 边的目标节点和端口
    data: { // 边的自定义数据
      id: edgeId, // 边的唯一标识符
      nextId: target, // 边的目标节点
      title: null, // 边的标题
    },
  }
  return edgeData
}

async function handleCopyExport() {
  try {
    await navigator.clipboard.writeText(state.exportData); // 将数据复制到剪贴板
    notification.info({
      message: '提示',
      description: '复制成功',
      duration: 0.8,
    })
  } catch (e) {
    console.log(e)
    notification.error({
      message: '提示',
      description: '复制失败，请用ctrl + c',
    })
  }
}
</script>

<style lang="less">
.editor {
  height: 100%;
  display: flex;
  flex-direction: column;
  background-color: #f0f0f0;

  .editor-header {
    height: 50px;
    background-color: #fff;
    border-bottom: 1px solid #ccc;
    display: flex;
    align-items: center;
    justify-content: center;
    padding: 0 20px
  }

  .editor-content {
    flex: 1;
    display: flex;
    overflow: hidden;
  }

  .editor-sider {
    width: 200px;
    background-color: #fff;
    border-right: 1px solid #ccc;
  }

  .editor-canvas {
    flex: 1;
    background-color: #fff;
  }
}

.node {
  width: 160px;
  height: 40px;
  padding: 10px;
  border: 1px solid #ccc;
  cursor: move;
  border-radius: 5px;
  display: flex;
  align-items: center;
  background-color: #fff;

  &:hover {
    background-color: #deeff5;
    border-color: #0099ff;
  }

  .node-icon {
    margin-right: 10px;
    font-size: 14px;
    color: #FFF;
    border-radius: 5px;
    width: 20px;
    height: 20px;
    display: flex;
    align-items: center;
    justify-content: center;


  }

  .node-title {
    font-size: 14px;
    overflow: hidden;
    /** 不换行 自动省略 */
    white-space: nowrap;
    text-overflow: ellipsis;
    max-width: 100px;
  }
}
</style>